tensor([[0.5500, 0.5500, 0.5500, 0.5500, 0.5500],
        [0.4500, 0.4500, 0.4500, 0.4500, 0.4500]], grad_fn=<RoundBackward1>)